Reference: TIS Kubo 2018

Factory method part is omitted.

Online Editor

https://plantuml-editor.kkeisuke.com

Sequence chart

Source for plantuml

@startuml
'0. Initialize
User -> main
activate main

main -> Config : new()
activate Config

main -> Experiment : new()
activate Experiment

main -> Transformer : new()
activate Transformer

main -> Trainer : new()
activate Trainer

main -> ModelAPI : new()
activate ModelAPI

main -> Storage : new()
activate Storage

main -> Experiment : set(tfm, trn, mapi)
main -> Experiment : run()


'1. Transform
Experiment -> Transformer : transform(train, test)

Transformer -> Feature : get_feature()

activate Feature
Feature -> Storage : load()
Feature -> Feature : create_features()
Feature -> Storage : save()
Feature -> Transformer : train, test
deactivate Feature

Transformer -> Experiment : train, test
deactivate Transformer

'2. Train
Experiment -> Trainer : train()
Trainer -> Model : fit()

activate Model
Model -> Model : fitImpl
Model -> Trainer : model
deactivate Model

Trainer -> Storage : save()
Trainer -> Experiment : model
deactivate Trainer

'3. Predict
Experiment -> ModelAPI : predict()
ModelAPI -> Model : predict()

activate Model
Model -> Model : predictImpl
Model -> ModelAPI : prediction
deactivate Model

ModelAPI -> Storage : save()
ModelAPI -> Experiment : None

deactivate ModelAPI
deactivate Storage

'4. Finish
Experiment -> main
deactivate Experiment
deactivate Config
deactivate main
main -> User

@enduml